package com.qcu.agent;

import dev.langchain4j.service.MemoryId;
import dev.langchain4j.service.SystemMessage;
import dev.langchain4j.service.UserMessage;
import dev.langchain4j.service.spring.AiService;
import dev.langchain4j.service.spring.AiServiceWiringMode;
import jakarta.annotation.Resource;
import reactor.core.publisher.Flux;

/**
 * @Name: MyComputer
 * @ClassName ShuGoAI
 * @Description:
 * @date 2025/4/24 18:06
 */
@AiService(
        wiringMode = AiServiceWiringMode.EXPLICIT,
//        chatModel = "qwenChatModel",
        streamingChatModel = "qwenStreamingChatModel",
        chatMemoryProvider = "chatMemoryProviderShuGo",
        contentRetriever = "contentRetrieverShuGo",
        tools = "appointmentTool"
)
public interface ShuGoAI {

    @SystemMessage(fromResource = "shugo-prompt-template.txt")
    Flux<String> chat(@MemoryId Long memoryId, @UserMessage String message);
}
